tensor([[0.4100, 0.4100, 0.4100, 0.4100, 0.4100],
        [0.5900, 0.5900, 0.5900, 0.5900, 0.5900]], grad_fn=<RoundBackward1>)